//
//  IMCellDisplayData.m
//  ChatMainDemo
//
//  Created by peiw on 16/1/21.
//  Copyright © 2016年 peiw. All rights reserved.
//

#import <UIKit/UIKit.h>
#import "IMCellDisplayData.h"
#import "IMChatDefines.h"

@implementation IMCellDisplayData

- (UIImage *)bubbleImage {
    if (self.direction == Send) {
        return [UIImage imageNamed:ChatBubbleFrom];
    }else {
        return [UIImage imageNamed:ChatBubbleTo];
    }
}

- (IMCellConfig *)chatCellConfig {
    IMCellConfig *config = [IMCellConfig new];
    if (self.direction == Send) {
        config.type = BubbleLeft;
    }else {
        config.type = BubbleRight;
    }
    return config;
}

@end
